package cc.ok200.muzhi;

import android.content.Context;
import android.util.Log;

import java.io.BufferedReader;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStreamReader;

/**
 * 存储工具类
 * 用于读写文件到应用私有目录或指定路径
 */
public class Storage {
    private static final String TAG = "Storage";
    private static Context sContext;

    /**
     * 初始化存储工具，必须在使用前调用一次
     * @param context 应用上下文
     */
    public static void init(Context context) {
        sContext = context.getApplicationContext();
    }

    /**
     * 写入内容到指定路径
     * 如果pathname不以"/"开头，如"a/b/c/123.txt"，则写入到"/sdcard/应用包名/a/b/c/123.txt"
     * 如果pathname以"/"开头，如"/sdcard/a/b/c/123.txt"，则直接按该路径写入
     *
     * @param pathname 文件路径
     * @param content 要写入的内容
     * @return 是否写入成功
     */
    public static boolean write(String pathname, String content) {
        if (sContext == null) {
            Log.e(TAG, "Storage未初始化，请先调用init()方法");
            return false;
        }

        if (pathname == null || content == null) {
            Log.e(TAG, "路径或内容不能为空");
            return false;
        }

        // 确定完整路径
        String fullPath;
        if (pathname.startsWith("/")) {
            // 以/开头的路径，按原路径处理
            fullPath = pathname;
        } else {
            // 不以/开头的路径，添加应用包名前缀
            fullPath = getBasePath() + "/" + pathname;
        }
        
        File file = new File(fullPath);

        // 确保目录存在
        File dir = file.getParentFile();
        if (dir != null && !dir.exists()) {
            if (!dir.mkdirs()) {
                Log.e(TAG, "创建目录失败: " + dir.getAbsolutePath());
                return false;
            }
        }

        // 写入内容
        try (FileOutputStream fos = new FileOutputStream(file)) {
            fos.write(content.getBytes());
            Log.d(TAG, "文件写入成功: " + fullPath);
            return true;
        } catch (IOException e) {
            Log.e(TAG, "文件写入失败: " + e.getMessage(), e);
            return false;
        }
    }

    /**
     * 从指定路径读取内容
     * 如果pathname不以"/"开头，如"a/b/c/123.txt"，则读取"/sdcard/应用包名/a/b/c/123.txt"
     * 如果pathname以"/"开头，如"/sdcard/a/b/c/123.txt"，则直接读取该路径
     *
     * @param pathname 文件路径
     * @param defaultValue 读取失败时返回的默认值
     * @return 文件内容，读取失败返回defaultValue
     */
    public static String read(String pathname, String defaultValue) {
        if (sContext == null) {
            Log.e(TAG, "Storage未初始化，请先调用init()方法");
            return defaultValue;
        }

        if (pathname == null) {
            Log.e(TAG, "路径不能为空");
            return defaultValue;
        }

        // 确定完整路径
        String fullPath;
        if (pathname.startsWith("/")) {
            // 以/开头的路径，按原路径处理
            fullPath = pathname;
        } else {
            // 不以/开头的路径，添加应用包名前缀
            fullPath = getBasePath() + "/" + pathname;
        }
        
        File file = new File(fullPath);

        // 文件不存在，返回默认值
        if (!file.exists() || !file.isFile()) {
            Log.d(TAG, "文件不存在: " + fullPath);
            return defaultValue;
        }

        // 读取文件内容
        StringBuilder content = new StringBuilder();
        try (BufferedReader reader = new BufferedReader(new InputStreamReader(new FileInputStream(file)))) {
            String line;
            while ((line = reader.readLine()) != null) {
                content.append(line).append("\n");
            }
            
            // 如果文件不为空，删除最后一个换行符
            if (content.length() > 0) {
                content.deleteCharAt(content.length() - 1);
            }
            
            Log.d(TAG, "文件读取成功: " + fullPath);
            return content.toString();
        } catch (IOException e) {
            Log.e(TAG, "文件读取失败: " + e.getMessage(), e);
            return defaultValue;
        }
    }

    /**
     * 获取应用基础路径（/sdcard/应用包名/）
     */
    private static String getBasePath() {
        // 使用应用的外部存储目录 (/sdcard/Android/data/应用包名/files)
        File externalFilesDir = sContext.getExternalFilesDir(null);
        if (externalFilesDir != null) {
            return externalFilesDir.getAbsolutePath();
        }
        
        // 如果获取不到外部存储目录，则使用内部存储
        Log.w(TAG, "无法获取外部存储目录，使用内部存储");
        return sContext.getFilesDir().getAbsolutePath();
    }
} 